package com.scf.common.utils;

import com.scf.common.vo.ValidationResult;
import org.springframework.util.CollectionUtils;

import javax.validation.ConstraintViolation;
import javax.validation.Validation;
import javax.validation.Validator;
import javax.validation.groups.Default;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;

/**
 * 使用hibernate-validate来验证，其中分为手动和自动校验。
 * 1）自动校验可以联合spring，使用@Valid注解，以及 BindingResult类来自动收集消息；
 * 2）手动校验，随时随地的调用工具类，校验指定类或注定属性是否符合规则，将错误封装。
 * @author sunchuanfu
 */
// https://www.cnblogs.com/duguxiaobiao/p/11533505.html
// https://www.jianshu.com/p/024892833ca9
public class ValidateUtil {
    private ValidateUtil() {
    }

    /**
     * 验证器
     */
    private static final Validator VALIDATOR = Validation.buildDefaultValidatorFactory().getValidator();


    /**
     * 指定分组，校验实体，返回实体所有属性的校验结果
     *
     * @param obj   校验实体
     * @param group 分组名称
     */
    public static <T> ValidationResult validateEntityByGroup(T obj, Class<?>... group) {
        Set<ConstraintViolation<T>> validateSet = VALIDATOR.validate(obj, group);
        return buildValidationResult(validateSet);
    }

    /**
     * 校验实体，返回实体所有属性的校验结果
     *
     * @param obj 校验实体
     */
    public static <T> ValidationResult validateEntity(T obj) {
        // Default.class是默认分组名称
        Set<ConstraintViolation<T>> validateSet = VALIDATOR.validate(obj, Default.class);
        return buildValidationResult(validateSet);
    }

    /**
     * 校验指定实体的指定属性是否存在异常
     *
     * @param obj          校验实体
     * @param propertyName 校验实体的属性
     */
    public static <T> ValidationResult validateProperty(T obj, String propertyName) {
        Set<ConstraintViolation<T>> validateSet = VALIDATOR.validateProperty(obj, propertyName, Default.class);
        return buildValidationResult(validateSet);
    }

    /**
     * 将异常结果封装返回
     */
    private static <T> ValidationResult buildValidationResult(Set<ConstraintViolation<T>> validateSet) {
        ValidationResult validationResult = new ValidationResult();
        if (!CollectionUtils.isEmpty(validateSet)) {
            validationResult.setHasErrors(true);
            Map<String, String> errorMsgMap = new HashMap<>();
            for (ConstraintViolation<T> constraintViolation : validateSet) {
                errorMsgMap.put(constraintViolation.getPropertyPath().toString(), constraintViolation.getMessage());
            }
            validationResult.setErrorMsg(errorMsgMap);
        }
        return validationResult;
    }

}
